
from nlp_tools.corpus.classify.dialogue_corpus_loader import DialogueCorpusLoader
from nlp_tools.tasks.classification.abc_model import ABCTaskModel
from sklearn.metrics import classification_report,f1_score

valid_data_path = r'/home/fanfanfeng/working_data/nlp_data/working/contraband/data/20220106/valid/all.csv'
valid_result_path = r'./valid_result'
model_save_path = '/home/fanfanfeng/project/gitee/nlp_tools/example/classification/train_model'
loaded_model = ABCTaskModel.load_model(model_save_path)
valid_data = DialogueCorpusLoader.load_data(valid_data_path,shuffle=False)
loaded_model.evaluate(valid_data,result_save_path=valid_result_path)